package com.stlm2.util;

import com.stlm2.dataSource.DBContextHolder;
import org.apache.commons.lang.math.NumberUtils;

/**
 * Author：fuchunyu
 * Date：2017/12/7
 * Description：
 */
public class DataSourceUtil {
    public static Boolean changeDataSource(String companyCode){
        if(NumberUtils.isDigits(companyCode)){
            int nub=NumberUtils.toInt(companyCode);
            switch (nub) {
                case 1:
                    //郑州公司
                    System.out.println(" 拦截器 指定 数据源 公司编号："+001);
                    DBContextHolder.setDBType(DBContextHolder.masterData);
                    return  true;
                case 2:
                    //永城公司
                    System.out.println(" 拦截器 指定 数据源 公司编号："+002);
                    DBContextHolder.setDBType(DBContextHolder.yongChengData);
                    return  true;
                case 3:
                    //郑州新天地公司
                    System.out.println(" 拦截器 指定 数据源 公司编号："+003);
                    DBContextHolder.setDBType(DBContextHolder.xtdData);
                    return  true;
                default:
                    System.out.println(" 拦截器 指定 数据源 不存在公司编号："+companyCode);
                    return false;
            }
        }else {
            System.out.println(" 拦截器 指定 数据源 未传入公司编号");
            //郑州公司
            DBContextHolder.setDBType(DBContextHolder.masterData);
            return  true;

        }
    }
}
